package com.atguigu.gmall.realtime.utils;

import com.alibaba.fastjson.JSONObject;
import com.atguigu.gmall.realtime.common.GmallConfig;
import org.apache.commons.beanutils.BeanUtils;

import java.sql.*;
import java.util.ArrayList;
import java.util.List;

/**
 * Author: Felix
 * Date: 2022/4/27
 * Desc: 操作phoenix工具类
 */
public class PhoenixUtil {

    private static Connection conn;

    public static void initConnection() throws Exception {
        //注册驱动
        Class.forName("org.apache.phoenix.jdbc.PhoenixDriver");
        //获取连接
        conn = DriverManager.getConnection(GmallConfig.PHOENIX_URL);
        //设置操作schema
        conn.setSchema(GmallConfig.PHOENIX_SCHEMA);
    }

    /**
     * 从phoenix表中从查询数据的方法
     * @param sql 执行查询语句
     * @param clz 将查询结果封装为指定类型
     * @return 查询结果
     */
    public static <T> List<T> queryList(String sql, Class<T> clz) {
        List<T> resList = new ArrayList<>();
        PreparedStatement ps = null;
        ResultSet rs = null;
        try {
            if (conn == null) {
                initConnection();
            }
            //获取数据库操作对象
            ps = conn.prepareStatement(sql);
            //执行SQL语句
            rs = ps.executeQuery();
            //获取元数据信息
            ResultSetMetaData metaData = rs.getMetaData();

            //处理结果集
            //    +-----+----------+
            //    | ID  | TM_NAME  |
            //    +-----+----------+
            //    | 15  | bbb      |
            //    | 16  | ffff     |
            //    | 17  | fdsaf    |
            //    +-----+----------+
            while (rs.next()) {
                //将查询出来的一条记录  封装为java中的一个对象
                T obj = clz.newInstance();
                //将查询结果赋值给对象的属性     列名----属性名
                for (int i = 1; i <= metaData.getColumnCount(); i++) {
                    String columnName = metaData.getColumnName(i);
                    Object columnValue = rs.getObject(i);
                    //借助BeanUtils工具类，给对象的属性赋值
                    BeanUtils.setProperty(obj,columnName,columnValue);
                }
                resList.add(obj);
            }


        } catch (Exception e) {
            e.printStackTrace();
        } finally {
            //释放资源
            if (rs != null) {
                try {
                    rs.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
            if (ps != null) {
                try {
                    ps.close();
                } catch (SQLException e) {
                    e.printStackTrace();
                }
            }
        }
        return resList;
    }

    public static void main(String[] args) {
        List<JSONObject> jsonList = queryList("select * from dim_base_trademark", JSONObject.class);

        System.out.println(jsonList);
    }
}
